﻿// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.Net.Http;
using AngleSharp.Dom.Html;
using Microsoft.AspNetCore.Identity.FunctionalTests.Account.Manage;

namespace Microsoft.AspNetCore.Identity.FunctionalTests.Account;

public class LoginWith2fa : DefaultUIPage
{
    public const string Path = "/Identity/Account/LoginWith2fa";

    private readonly IHtmlFormElement _twoFactorForm;
    private readonly IHtmlAnchorElement _loginWithRecoveryCodeLink;

    public LoginWith2fa(HttpClient client, IHtmlDocument loginWithTwoFactor, DefaultUIContext context)
        : base(client, loginWithTwoFactor, context)
    {
        _twoFactorForm = HtmlAssert.HasForm(loginWithTwoFactor);
        _loginWithRecoveryCodeLink = HtmlAssert.HasLink("#recovery-code-login", loginWithTwoFactor);
    }

    internal async Task<Index> Send2FACodeAsync(string twoFactorKey)
    {
        var code = EnableAuthenticator.ComputeCode(twoFactorKey);

        var response = await Client.SendAsync(_twoFactorForm, new Dictionary<string, string>
        {
            ["Input_TwoFactorCode"] = code
        });

        var goToIndex = ResponseAssert.IsRedirect(response);
        Assert.Equal(Index.Path, goToIndex.ToString());
        var indexResponse = await Client.GetAsync(goToIndex);
        var index = await ResponseAssert.IsHtmlDocumentAsync(indexResponse);

        return new Index(Client, index, Context.WithAuthenticatedUser());
    }

    internal async Task<LoginWithRecoveryCode> ClickRecoveryCodeLinkAsync()
    {
        var goToLoginWithRecoveryCode = await Client.GetAsync(_loginWithRecoveryCodeLink.Href);
        var loginWithRecoveryCode = await ResponseAssert.IsHtmlDocumentAsync(goToLoginWithRecoveryCode);

        return new LoginWithRecoveryCode(Client, loginWithRecoveryCode, Context);
    }
}
